#include "task_queue.h"
#include "gmock/gmock.h"

namespace simpleThread {
    class TaskQueueMock : public simpleThread::TaskQueue {
    public:
        MOCK_METHOD(std::function<void()>, pull, (), ());
        MOCK_METHOD(unsigned long, size, (), (override));
    };
}